package com.jiinfo.util;

import java.io.File;
import java.io.FileOutputStream;
import java.io.OutputStream;
import java.text.SimpleDateFormat;
import java.util.ArrayList;
import java.util.Date;
import java.util.List;

import org.apache.poi.ss.util.CellRangeAddress;
import org.apache.poi.xssf.usermodel.XSSFCell;
import org.apache.poi.xssf.usermodel.XSSFCellStyle;
import org.apache.poi.xssf.usermodel.XSSFFont;
import org.apache.poi.xssf.usermodel.XSSFRichTextString;
import org.apache.poi.xssf.usermodel.XSSFRow;
import org.apache.poi.xssf.usermodel.XSSFSheet;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;

import com.jfinal.kit.PathKit;

/**
 * 
 * @author mengfj 2016年4月9日
 *
 */
public  class ExportUtil {
	
			
	public void export(XSSFWorkbook workbook, int sheetNum,
			String title,String sheetTitle ,String[] rowName, List<Object[]>  dataList,
			OutputStream out) throws Exception{
		try{
			XSSFSheet sheet = workbook.createSheet();
			workbook.setSheetName(sheetNum, sheetTitle);
			// 产生表格标题行
	        XSSFRow rowm = sheet.createRow(0);
	        XSSFCell cellTiltle = rowm.createCell(0);
	        
	        //sheet样式定义【getColumnTopStyle()/getStyle()均为自定义方法 - 在下面  - 可扩展】
	        XSSFCellStyle columnTopStyle = this.getColumnTopStyle(workbook);//获取列头样式对象
	        XSSFCellStyle style = this.getStyle(workbook);					//单元格样式对象
	        
	        sheet.addMergedRegion(new CellRangeAddress(0, 1, 0, (rowName.length-1)));  
	        cellTiltle.setCellStyle(columnTopStyle);
	        cellTiltle.setCellValue(title);
    		
			// 定义所需列数
			int columnNum = rowName.length;
			XSSFRow rowRowName = sheet.createRow(2);				// 在索引2的位置创建行(最顶端的行开始的第二行)
			
			// 将列头设置到sheet的单元格中
			for(int n=0;n<columnNum;n++){
				XSSFCell  cellRowName = rowRowName.createCell(n);				//创建列头对应个数的单元格
				cellRowName.setCellType(XSSFCell.CELL_TYPE_STRING);	
				//设置列头单元格的数据类型
    			XSSFRichTextString text = new XSSFRichTextString(rowName[n]);
    			cellRowName.setCellValue(text);									//设置列头单元格的值
    			cellRowName.setCellStyle(columnTopStyle);						//设置列头单元格样式
    		}
			
    		//将查询出的数据设置到sheet对应的单元格中
			for(int i=0;i<dataList.size();i++){
				
				Object[] obj = dataList.get(i);//遍历每个对象
				XSSFRow row = sheet.createRow(i+3);//创建所需的行数
				
				for(int j=0; j<obj.length; j++){
					XSSFCell  cell = null;   //设置单元格的数据类型
					cell = row.createCell(j,XSSFCell.CELL_TYPE_STRING);
					if(!"".equals(obj[j]) && obj[j] != null){
						cell.setCellValue(obj[j].toString());						//设置单元格的值
					}
					cell.setCellStyle(style);									//设置单元格样式
				}
			}
			//让列宽随着导出的列长自动适应
			for (int colNum = 0; colNum < columnNum; colNum++) {
	            int columnWidth = sheet.getColumnWidth(colNum) / 256;
	            for (int rowNum = 0; rowNum < sheet.getLastRowNum(); rowNum++) {
	                XSSFRow currentRow;
	                //当前行未被使用过
	                if (sheet.getRow(rowNum) == null) {
	                    currentRow = sheet.createRow(rowNum);
	                } else {
	                    currentRow = sheet.getRow(rowNum);
	                }
	                if (currentRow.getCell(colNum) != null) {
	                    XSSFCell currentCell = currentRow.getCell(colNum);
	                    if (currentCell.getCellType() == XSSFCell.CELL_TYPE_STRING) {
	                        int length = currentCell.getStringCellValue().getBytes().length;
	                        if (columnWidth < length) {
	                            columnWidth = length;
	                        }
	                    }
	                }
	            }
	            if(colNum == 0){
	            	sheet.setColumnWidth(colNum, (columnWidth-2) * 256);
	            }else{
	            	sheet.setColumnWidth(colNum, (columnWidth+4) * 256);
	            }
	        }

		}catch(Exception e){
			e.printStackTrace();
		}
		
	}
	
	/* 
	 * 列头单元格样式
	 */    
  	public XSSFCellStyle getColumnTopStyle(XSSFWorkbook workbook) {
  		
  		  // 设置字体
    	  XSSFFont font = workbook.createFont();
    	  //设置字体大小
    	  font.setFontHeightInPoints((short)11);
    	  //字体加粗
    	  font.setBoldweight(XSSFFont.BOLDWEIGHT_BOLD);
    	  //设置字体名字 
    	  font.setFontName("宋体");
    	  //设置样式; 
    	  XSSFCellStyle style = workbook.createCellStyle();
    	  //设置底边框; 
    	  style.setBorderBottom(XSSFCellStyle.BORDER_THIN);
    	  //设置左边框;   
    	  style.setBorderLeft(XSSFCellStyle.BORDER_THIN);
    	  //设置右边框; 
    	  style.setBorderRight(XSSFCellStyle.BORDER_THIN);
    	  //设置顶边框; 
    	  style.setBorderTop(XSSFCellStyle.BORDER_THIN);
    	  //在样式用应用设置的字体;  
    	  style.setFont(font);
    	  //设置自动换行; 
    	  style.setWrapText(true);
    	  //设置水平对齐的样式为居中对齐;  
    	  style.setAlignment(XSSFCellStyle.ALIGN_CENTER);
    	  //设置垂直对齐的样式为居中对齐; 
    	  style.setVerticalAlignment(XSSFCellStyle.VERTICAL_CENTER);
    	  
    	  return style;
    	  
  	}
  	
  	/*  
	 * 列数据信息单元格样式
	 */  
  	public XSSFCellStyle getStyle(XSSFWorkbook workbook) {
	  	  // 设置字体
	  	  XSSFFont font = workbook.createFont();
	  	  //设置字体大小
	  	  //font.setFontHeightInPoints((short)10);
	  	  //字体加粗
	  	  //font.setBoldweight(XSSFFont.BOLDWEIGHT_BOLD);
	  	  //设置字体名字 
	  	  font.setFontName("Courier New");
	  	  //设置样式; 
	  	  XSSFCellStyle style = workbook.createCellStyle();
	  	  //设置底边框; 
	  	  style.setBorderBottom(XSSFCellStyle.BORDER_THIN);
	  	  //设置左边框;   
	  	  style.setBorderLeft(XSSFCellStyle.BORDER_THIN);
	  	  //设置右边框; 
	  	  style.setBorderRight(XSSFCellStyle.BORDER_THIN);
	  	  //设置顶边框; 
	  	  style.setBorderTop(XSSFCellStyle.BORDER_THIN);
	  	  //在样式用应用设置的字体;  
	  	  style.setFont(font);
	  	  //设置自动换行; 
	  	  style.setWrapText(false);
	  	  //设置水平对齐的样式为居中对齐;  
	  	  style.setAlignment(XSSFCellStyle.ALIGN_CENTER);
	  	  //设置垂直对齐的样式为居中对齐; 
	  	  style.setVerticalAlignment(XSSFCellStyle.VERTICAL_CENTER);
	  	 
	  	  return style;
  	
  	}
  	
  	public static void main(String[] args) throws Exception {
  		//FileUtil.delAllFile(PathKit.getRootClassPath()+File.separator+"excel");
  		//File file = new File(PathKit.getRootClassPath()+File.separator+"excel"+File.separator+"123.xlsx");
  		//String filePath=PathKit.getWebRootPath()+File.separator+"excel";
  		//System.out.println(filePath);
  		System.out.println(PathKit.getWebRootPath());
  		/*OutputStream out = new FileOutputStream(file);
  	    String[] rowsName = new String[]{"序号","货物运输批次号","提运单号","状态","录入人","录入时间"};  
	  	List<Object[]>  dataList = new ArrayList<Object[]>();  
	  	Object[] objs = null;  
	  	for (int i = 0; i < 3; i++) {  
	  	    objs = new Object[rowsName.length];  
	  	    objs[0] = i;  
	  	    objs[1] = "aaa";  
	  	    objs[2] = "bbb";  
	  	    objs[3] = "ccc";  
	  	    objs[4] = "ddd";  
	  	    SimpleDateFormat df = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");  
	  	    String date = df.format(new Date());  
	  	    objs[5] = date;  
	  	    dataList.add(objs);  
	  	}  
	  	ExportUtil ex = new ExportUtil();  
	  	XSSFWorkbook workbook = new XSSFWorkbook();
	  	ex.export(workbook, 0, "123","总计", rowsName, dataList, out);  
	  	ex.export(workbook, 1, "456","明细", rowsName, dataList, out); 
	  	workbook.write(out);
		out.close();*/
	}
}